How to write this function in one line of code